package db

import (
	"database/sql"
	"fmt"
	gconfig "funlog-gserver/config"
	model "funlog-gserver/model/db"
	"log"
	"strconv"
	"sync"

	"gorm.io/driver/mysql"
	"gorm.io/gorm"
	"gorm.io/gorm/logger"
	"gorm.io/gorm/schema"
)

var SqlDB *sql.DB
var DB *gorm.DB
var once sync.Once

//InitDB 初始化数据库
func InitDB() {
	once.Do(func() {
		SqlDB, DB = openDB()
	})
}

//openDB 打开数据库
func openDB() (*sql.DB, *gorm.DB) {
	// gconfig.Cnf
	dbPort, err := strconv.ParseInt(gconfig.Cnf["DB_PORT"], 10, 32)
	if err != nil {
		log.Fatal("DB_PORT conv error")
	}
	dbAddr := gconfig.Cnf["DB_ADDR"]
	dbUser := gconfig.Cnf["DB_USER"]
	dbPwd := gconfig.Cnf["DB_PASSWORD"]
	dbName := gconfig.Cnf["DB_NAME"]
	dbCharset := gconfig.Cnf["DB_CHARSET"]
	// dbTimeZone := gconfig.Cnf["DB_TIMEZONE"]
	mylogLevel, err := strconv.Atoi(gconfig.Cnf["DB_LOGLEVEL"])
	if err != nil {
		mylogLevel = (int)(logger.Info)
	}
	dsn := fmt.Sprintf("%s:%s@tcp(%s:%d)/%s?charset=%s&parseTime=True&loc=Local", dbUser, dbPwd, dbAddr, dbPort, dbName, dbCharset)
	gormDB, err := gorm.Open(mysql.New(mysql.Config{
		DSN:                       dsn,   // DSN data source name
		DefaultStringSize:         256,   // string 类型字段的默认长度
		DisableDatetimePrecision:  true,  // 禁用 datetime 精度，MySQL 5.6 之前的数据库不支持
		DontSupportRenameIndex:    true,  // 重命名索引时采用删除并新建的方式，MySQL 5.7 之前的数据库和 MariaDB 不支持重命名索引
		DontSupportRenameColumn:   true,  // 用 `change` 重命名列，MySQL 8 之前的数据库和 MariaDB 不支持重命名列
		SkipInitializeWithVersion: false, // 根据当前 MySQL 版本自动配置
	}), &gorm.Config{
		Logger: logger.Default.LogMode((logger.LogLevel)(mylogLevel)),
		NamingStrategy: schema.NamingStrategy{
			TablePrefix:   "tb_",
			SingularTable: true,
		},
	})
	if err != nil {
		log.Fatal(err)
	}
	sqlDB, err := gormDB.DB()
	// SetMaxIdleConns 用于设置连接池中空闲连接的最大数量。
	sqlDB.SetMaxIdleConns(10)
	// SetMaxOpenConns 设置打开数据库连接的最大数量。
	sqlDB.SetMaxOpenConns(100)
	if err != nil {
		log.Fatal(err)
	}
	log.Println("db ", dbName, "connected")
	sqlDB.Ping()
	if gconfig.Cnf["DB_AUTOMIGRATE"] != "" {
		gormDB.AutoMigrate(&model.User{}, &model.Application{}, &model.Team{}, &model.TeamRelation{})
	}
	return sqlDB, gormDB
}
